screening sinkhorn algorithm
Screening Sinkhorn Algorithm for Regularized Optimal Transport
We introduce in this paper a novel strategy for efficiently approximating the Sinkhorn distance between two discrete measures. After identifying neglectable components of the dual solution of the regularized Sinkhorn problem, we propose to screen those components by directly setting them at that value before entering the Sinkhorn problem. This allows us to solve a smaller Sinkhorn problem while ensuring approximation with provable guarantees. More formally, the approach is based on a new formulation of dual of Sinkhorn divergence problem and on the KKT optimality conditions of this problem, which enable identification of dual components to be screened. This new analysis leads to the Screenkhorn algorithm. We illustrate the efficiency of Screenkhorn on complex tasks such as dimensionality reduction and domain adaptation involving regularized optimal transport.
Reviews: Screening Sinkhorn Algorithm for Regularized Optimal Transport
This paper proposes a reformulation of the dual of the entropy regularized wassserstein distance problem that is amenable to screening techniques. Such techniques allow to reduce the dimension of optimization problems hence reducing the computational costs. Here static screening rule is proposed, meaning that the variables are screened before running the solver which is here an L-BFGS-B quasi-Newton method. Two screening techniques are proposed, either using a fixed threshold or a fixed budget. The latter appearing easier to use. A theorem quantifying the error on the orignal problem induced by approximation and screening is provided.
Screening Sinkhorn Algorithm for Regularized Optimal Transport
Alaya, Mokhtar Z., Berar, Maxime, Gasso, Gilles, Rakotomamonjy, Alain
We introduce in this paper a novel strategy for efficiently approximating the Sinkhorn distance between two discrete measures. After identifying neglectable components of the dual solution of the regularized Sinkhorn problem, we propose to screen those components by directly setting them at that value before entering the Sinkhorn problem. This allows us to solve a smaller Sinkhorn problem while ensuring approximation with provable guarantees. More formally, the approach is based on a new formulation of dual of Sinkhorn divergence problem and on the KKT optimality conditions of this problem, which enable identification of dual components to be screened. This new analysis leads to the Screenkhorn algorithm.